Excercise 2 and changes on plot 1