Clean up notebook more, more plots