Fixing the model in the code and adding readme