fix: changing all PRNGKey to key because its going to be deprecated soon

fix: changing all PRNGKey to key because its going to be deprecated soon #22