Code highlight by filename

PR for #253

More testing is welcome. I didn't really see a good way to add tests for this change.