Import os module before using it