When should you stop swaddling your baby? Do you wean off slowly or is there a time you just suddenly stop swaddling?

I stopped swaddling my kids when they turned 2-3 months old, as they can sleep better around this period. Also, it will be danger if you swaddle your baby when they can flip over and unable to flip back (which is around 4months). You can do it in phases, i.e. swaddle one arm only for a couple of days and slowly let the both arms out. Please refer to the picture attached.