nicer plot and .sub changed