Adding some UI design