Wrap scatter

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>
Optional extensions for savefig

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>