Added os import statement per issue #1