Some minor changes to the plotting stuff like axis