Add missing 'os' import