Add missing fields for Task model