Add editTask function in