Add stuff to main()