Updated the model and script for better documentation