Fixed minor problems with code and added a figure with abstract representation of the model