Add plots and updated code