use matplotlib layout stuff for nicer plots