Add wind_direction function