NBC: Jay-Z and Yo Gotti dismiss federal lawsuit against Mississippi prison after upgrades