Added code and plots assignment