Added highlighting for comments.
Fixed syntax highlighting when loading code from storage.