Added a functional UI