If Statement Python Multiple Returns