Model

Added classes to the model