This model is created using the code from this repo