adapt for 1 and 2 images.