Update a_step-4.py