Sliding Window Code Python