add tests for return types