Unit 2 done, add more comments