Create code.py

Created in python.