Update documentation of code