Update main.py

Made it a bit less noobier. Script is now strictly business.