added highlighting for other code sections

IT WORKS!!! xD