Dodali smo funkcijo main.