Added return to main()