Improve Wrf_model class