Upgraded Pong AI to consider winning prob and avg points values