implemented return statement
corrected order of functions and parameters generated output of new test cases