Tidy
Add test