Adding class with main mathod