added small changes to UI