John Chang, founder and CTO of Willow, told CNNMoney the product was inspired by his wife and three children.

Chang hopes his product enables more women to breastfeed, because the pumps are discreet, simple to use and quiet. In trials, mothers wore them on train commutes and during conference calls.

The CDC recommends children breastfeed until they’re at least 12 months, but while 81% of infants born in the U.S. start breastfeeding, just 50% of them still do at six months.

The companion app is a convenient addition to the system. It tracks the amount of milk pumped from each breast, how frequently someone pumps, and how long sessions last. Instead of having to log activity in a notebook or breast pump app, Willow does it for you.

Other baby tech displayed at CES includes Snoo, an infant bed that prevents babies from rolling on their stomachs, which can put their lives at risk. Created by pediatrician and author Dr. Harvey Karp, the bassinet is made for newborns to 7-month-old babies.

An audio sensor listens for noise, and the bed automatically moves around when it hears a child start to cry. Parents can manually turn the bed on, too.

Snoo also comes with swaddling that zips up the front so the baby can keep its limbs close and comfortable while it sleeps.