fixed some stuff for nicer plots