Update main.py

Adding ability to count winnings toward computer