If you were correct writing No OS Module then it is possible that you had used caps letters - it is incorrect. In Python os and OS are different modules and there is no module in standard library named OS.
So, try correcting it to import os.
Another solution is to check your python environment used by ArcGIS. Here is some further discussion on Esri forum. One of solutions from that thread:
SOLUTION: The ntpath.pyc module, necessary for importing the os
module, is not in the C:\Program Files\ArcGIS\bin path. To resolve
this, if it happens to you, set the PYTHONPATH Environment variable to
include the path to the same directory where PythonWin finds the
ntuser.pyc module. You can determine this easily by going to the
Interactive Window in PythonWin or IDLE and typing: import os;
os.path. It will typically be in the following directory
C:\Python25\Lib or something like that. Include that path as a value
in your PYTHONPATH Environment variable (separate paths with
semicolons) and see if this test script will run as a Script Tool in
the ArcGIS Toolbox. It should work. It did for me.
Apparently, when running in PythonWin, the loader searches in the same
directory where the python.exe is found and doesn't use (need)
PYTHONPATH to resolve the reference. When running under the python.exe
in the C:\Program Files\ArcGIS\bin it does need PYTHONPATH to resolve
the reference. On the other hand, when running in PythonWin, the
loader does use the PYTHONPATH Environment variable to resolve the
arcgisscripting reference.
Hope it helps you!