savefig takes figure as optional parameter

Signed-off-by: Matthias Kümmerer <matthias@matthias-k.org>