Fixed Bugs in: Controller