Add tests

:+1: thanks