Addressing PR comments about formatting and extending dist_kv_test