Putting back in necessary os import