Python Def If Return