Add tests for SRP