Add docs. Proper docs. with example too.

``` python
"""
 Brief description of the class
 input params
 output params
 Example
"""