Add tests
Add new flags