added new tests in main()