Added better code highlighting