Water Jet Problem In Python