updated notebook for printing more plots