Javascript solution to problem 1