Added code for main class and more