fix plots to use consistent colors + labels like other lessons, remove unnecessary material