organizing, small change to saving image