Add doctrings
Add doctrings