continue solving problem 5