Store Function In Variable Python