Yes, it gets better, but the season has been criticized for lacking narrative logic, even more than when RTD was the showrunner the first time. So, some things will make similar not-sense* as "Space Babies". But the character stuff and dramatic moments continue to be spot-on.
* This was not...