Further work on the models and updated README