Update __init__
Add some docstrings