Add tests for add_timestamp and timestamp_key