Modify python-mode