Add back notes highlighting