Updated the Main class to be able to test further aspects of the project