Add OS check