Fix highlighting colors on files and address the undefined error. Closes #477