added more plots
also changed notebook