Fix a little bugs
Update docs