update to classes

update to classes