removed the puts and added the return