upravené malé problémy + zmenená verzia