improved efficiency of solution to problem 4