add assignment code and plots