Added --count arg to run.py
feat(add --count argument to run.py)