Dodao neke sitne funkcionalnosti u main