Added classes for models