Need for Types

hey @pivanov  @IraGR, the code is missing types which makes it difficult to interpret how the code will work and what is it providing to the user, so I think making types for the project can be great help for the developers using this package. 

**If you guys agree than I will write the types and create a pull request.**