módosítottam a main-t!