Problem solving - problem 2 solved