Added code in the documentation