continue solving problem 6