Adding classes and initializing within init() function