Add python-mode support