solution for baby_steps