Task edit UI fixes