Update a_step-6.py