Fixed things in main().