small changes to replay tool