Cleaned up controller

Signed-off-by: A.Prooks <aprooks@live.ru>