Moved argument check into main()