Update funkcji Main

Nie wiem do końca czy tak ma to działać, ale w razie problemów zawsze można zmodyfikować