update the test/main part....