Improved plot in intro notebook