Finished controller code!

Controller code finished. All finished actually!