Add OS_TYPE logic