Add in python code, fix text