Week 1 assignment, fixed plot1