added some TODO comments for the students