Adding proper code highlight