updated classes in model -amelia