How To Make A Function Return A Variable In Python