Added branch highlighting functionality.