Update the plots to clean up axis and stuff