Remove unittest.main()