Added a test for returning a value