Added documentation, cleaned up tests and added return types
Merge pull request #1 from cavneb/feature/enhance_readme

Feature/enhance readme