importing task_types+ fitting other code