Add tests for registration