Add tests
Add main
Add tests