Improve comments, changed calling of main()
Fix minor typo - using return instead of break