Update main.py

näköjään edellinen versio oli siistimätön ja kommentoimaton, joten lisätty em. asioita