#19 Add tests