Cleaning up and fixing mistakes, solve problem 4