modify ass2 task1