update plots and code

update plots and code