Adding classes into main program