Add jedi mode for python