A mum whose baby spent weeks in neonatal care welcomes a law which will give parents extra time off.