Added highlighting to the code