Improvements
* Search definition in current file first.
* Improve the regex used to search for 'from xxx.xx import xx' pattern