Added include for iostream library