Add doc strings
Update documentation