I believe the answer is Piaget.
Piaget divided the children's development into Four stages of Development.
He based the stages on the assumption that our brain develop a certain capability/capacity as we grow older, and our cognitive ability also increased alongside it.