Change initialize method