Added code to documentation