Znowu troche refaktoring, ale dodalem rowniez feature'y.