Made keyword and comment highlighting work better in the REPL