Add code and plots