working on adding classes