working on adding unit tests now