updated classes as needed