Ui

Add Controller component to make UI more cohesive.