Water Jug Problem Ai Python