Add tests for returned object