Remove unnecessary lines and refactor hard to read and messy code in main().
Remove unnecessary lines and refactor hard to read and messy code in main().