Add proper code highlighting