Cleanup + adding docs for `countSteps`