Add support for jedi 0.8

The "FUNCTION" keyword has been replaced by "DEF"