task 2.2 updated, thanks to Huy's good comments!