Improve Syntax Highlighting for Python/JS