Addressing #23

A few sentences describing the changes proposed in this pull request.

Type of change

Please delete options accordingly to the description.

[ ] Bug fix (non-breaking change which fixes an issue)
[x] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
[ ] This change requires a documentation update

[ ] I have read the CONTRIBUTING docs and have a developer setup (especially important are pre-commitand pytest)
[ ] My code is properly tested
[ ] My code follows the style guidelines
[ ] My code does not contain unused elements. If so, I documented why it needs to.
[ ] My changes generate no new warnings
[ ] I have performed a self-review of my own code
[ ] I have commented my code, particularly in hard-to-understand areas
[ ] I have made corresponding changes to the documentation
[ ] This pull request is ready to be reviewed
[ ] My branch is up-to-date with master (with use of rebase if needed)
[ ] If the PR is ready and there are multiple commits, I have squashed them and force-pushed