Adding some python code