Fixed bug with problem 5 main function