Remove unnecessary lines in code sample