Use A Custom Function Python