Added a return to the function to return itself