Add initialization code