Update for shell highlighting