solves one test

and the second test