Tests: Add tests for public methods