Update incomplete_code.py

addd Train2