Add/improve type hints for functions in `simulate_data.py`

[drbenvincent]

We currently don't have any type hinting for any of the functions in simulate_data.py. It would be good to add this. That will also act as an improvement to the auto generated API docs.

[Ishaanjolly]

Can I deal with this issue? Seems like I'm onto documentations atm.