Add tests for new API classes