Create main_code.py

Added all the prototype code.