task5 fixes