Remove extra print, correct return statement