Made UI changes for task feature