Fixing code highlighting