feat(added --count arg to run.py)

Closes #35 and #36