Added game size to simplify code slightly

Tested on python3 works
Tested on python2 Does not work.