add more tests, take out print statement