Added todos for part b of the project