notebooks for reward decomposition