Simple collecting game with python/pygame