Added controller
Merge pull request #9 from alfeko/UI

Added controller