Ooops forgot the plotting stuff