Added controller
Finished!