add a simple count test as well