trying to solve problems with tests - no luck :(