Update main.py

Well i think it's ready. but it's probably all sorts of wrong
it has been helping me