Water Jug Problem Code In Python