changes in ui
Adding functionality, so that program running