Beautified the main()