Subarray Program In Python