To return or not to return... and I don't really know Python