added some of the classes need to complete iteration2