Renamed main function to assignment4
Prints proper output