[Cyntain] More work on the main.py