added test in main