Updated docs for code