Added iostream include for ubuntu