I found the mechanic simple but fun. To improve it you could consider to make it clearer if the input after interacting with the baby was successful or not. Right now if the same key shows again that you have to press, it's not very clear if it's because you succeeded on the last one, or didn't press it yet. Also, when I pressed an arrow key for movement instead of WASD I think it broke the game and I may have had to start over.