problem 4 working