Add update_wind function