- Updated plots to match with notebook