Removed os module import