Ministers said they wanted to tackle difficulties parents face when their baby is in neonatal care such as having to return to work.