Majhni popravki

Poprabljena napaka, dodana pomožna funkcija in komentarji