@dlonie spotted that doc string needed to be updated
@dlonie noted we can cleanup import statement
cleaned up import a bit too much, still needed count