Added More styling and updated main.py