Update main.py

sys.argv + kontrola vstupov