adding assignment 4 main class