Good evening. Yes, Ol' Nebbi of Babylon took Judah captive about 602 BC.
Referencing Daniel, we remember how the Persians overran Babylon. The
Persian king Cyrus who Daniel mentions told all the captives from the
nations they could go home if they wanted.
   So, yes, the first return of the Jews dispersed throughout the
Persian empire was about 538 BC, led by Zerubbal, a descendant of
David.
   About eighty years later, around 458 BC, Ezra the scribe was moved
by God to lead a return to rebuild God's house in Jerusalem. Remember
that fuss >
   The final return was around 458 BC under Nehemiah the ordinary man
who God told to rebuild the walls of Jerusalem. It's likely the King
allowed him to do this because Queen Esther was sitting next to the
king and elbowed him in the ribs!
   Anyway, see this page:
https://israelmyglory.org/article/the-exiles-return/
Maranatha, Hasten that day.
      Lewis B