Forgot a pretty important return function

Added it back in now...oops