Removed unnecessary import of the os module.