Updated documentation of code