Documentation update + bug fix