Add comments
Update documentation