Доделал код.