Added some more UI
Added more UI