updates to classes and unit test
update to tests