Added class and ref keywords to check highlighting. Does not work well yet.