working on adding tests back