Updated the code docs