Add tests for challenge