updates to plots and code