added test in main
added test in main