Sleeping on tummy

My son will be 4 months on Saturday. He’s finally teaching himself to roll over onto his belly. He’s still swaddled with his arms out at night but just tonight right now he kept looking like he was trying to pull himself over waking up so much! I brought him into my bed and tried to sleep with him like that and he wouldn’t sleep still trying to roll over. So i put him on his belly and he instantly calmed down and knocked out. Should i take him out the bassinet and put him in this crib now since he wants to sleep belly down !? Noooo ideaaaa!