small changes to savefig