Add an argument to a method, replace puts with return