Hey! At this age, it would be recommended for babies to sleep on their back, even though they would be waking up frequently. This helps them to breathe well. Babies need to be strong enough to roll on their back and stomach for which, you may wait for few more months!