Add proper OS handling