Slight modifications in IO.cs