Minor - changes to the controller extends class.