UI cleanup for tasks