Adding test for initialize method