Finally fixed the bug with the highlighting of the function completely