working on adding some tests