Switch from print to return in methods