Refaktor na MVVP + zmiany UI