- added testReturn to unit tests