Python Store Function In Variable