Removed unnecessary import of os