Updating code highlighting script