Fix code blocks in documentation