Create small_code_but_do_same.py

the finall edit