updated solution to problem 16