Added return for main()