starting problem 5
starting problem 5
fix tests