Added few docstrings
Separate init into two methods
Added test