Implemented return statements and function calls