Adding testing to main()