Add tests
Add lenses