Adding unit testing in place of main.go