Modified main.py to only run CountsGenerator.run() once instead of twice.