python-mode improvements