Updated controller to reflect module changes.