Dodani testovi i male izmjene metoda