Flatten one more time, get rid of __init__ and just have a main.py
Add ct()
Fix for new structure