updating references to updated classes