Added plots and code