Fix bugs in main.py and clean up Cells