A pool game in pygame