problem 2 working