improved code highlighting