added main to test
added main to test