add code and plots