Self created python code