Update incomplete_code.py

add main class