problem 4 working ish