Add code to docs