Update IM_test_part_1.py