main_test.go added