expected output added to code section