Updated code blocks in documentation