added color highlighting functionality