Python Code To Flatten Out Over Time