count should only print a number
update tests to not write to stdout
updates to tests