Make jedi work with imenu