added color plotting