Added tests
Oops wrong file