Water Jug Problem In Python Simple Code