Fixed bug to allow assignment1_main to run