Fixing Controller class

Fixing Controller class