Added missing os import