Update main.py

CSS styling