nicer plots, everything works perfectly now