Update take_last_while example