Added figures and other tweaks to the description of the model structure