Add tests for render method