Advice for newborn that can roll

My two week old can roll over from back to stomach. She’s very strong she can pick her head up but we’ve been letting her sleep in the rock n play. But she doesn’t sleep soundly in there because she scooches down and then gets mad that she can’t stretch and it’s a long night with her in that thing. So we were going to put her in the pack n play tonight. But then she rolled onto her stomach. I’m what should I do? What do you guys have your babies sleep in at first??? Thanks