Допилил функционал