refaktorovana funkcia

plus opravene nejake chyby v nej...  este sa tomu asi budem venovat lebo
sa mi stale nepaci, ale podmienky uz splna