fix size
recreate classes in main.py