Add init code