Improve syntax highlighting in edts-shell