Adding some UI