Plotting assignment 1: final modifications