Fix comment
Add main controller class