Transferred functions in *.py files and readjusted code

[review-notebook-app[bot]]

Check out this pull request on ReviewNB: https://app.reviewnb.com/OpenSourceEconomics/dc-egm/pull/10

You'll be able to see notebook diffs and discuss changes. Powered by ReviewNB.