Create logic.py