Python Create A Function To Store Variable