Adding models to support plotting