Adds function to init python file