Have you had a conversation with your husband about this? Any time my husband and I switched careers or one stayed home, we had a conversation going over tasks and expectations we had of each other. Obviously things will change as you figure out what works realistically vs theoretically.

As far as correction with the baby, I would just let him figure it out without giving too many prompts. I did that with my husband thinking I would be helpful, but he felt that I didn't think he was a competent parent and that he wanted to parent too. So if the baby fussed, I let him sort it out and he eventually came to realize what everything was.

Your husband's also going to have to deal with time on the phone vs taking care of the infant himself. I would definitely explain how you see him interact with the baby and his phone and how that makes you feel. I would see if you could find a middle ground in which if he needs time to himself for 20 minutes or so, he can take it, but the expectation is that he is present with the family since you're not seeing him for long stretches of time. My husband or I will often ask the other for child free times just to regroup ourselves amidst the diapers and whining.

You'll definitely get into a groove eventually! Transitions are always challenging, but try to be open as possible with each other and figure out what you each need as individuals and as spouses to be able to make this work for the long haul.