Create code.py

Main code