use test.go for testing