Tidy
Add tests