Update main.py

Adding colours and label frame