Trapping Rain Water (Python)