Update code highlighting in the README