updated solution to problem 6