Modified 'return' highlighting