Shell completion for python