Add tests for the other methods