Add tests

;)