Cleaning up

Подчистил код от отладочных комментариев.
Допилил функционал