problem2 working