Pool game attempt in Python