Waist trainers!

Cali☀️
I'm due in August and have never worn a waist trainer before but I want to buy one for after baby. Which kind do you recommend? I want the kind that Kim kardashian or Amber Rose uses or something similar