Update example code to run inside main
#328
+8
−6
Open
main
#328