Updated code snippet style for docs