Two improvemnts.

Added `import os`. FOrgot about it.
Added a comment.