fix image output; change to iteration 2