Task cell UI fixes