Fix file formatting issues. File is now nicer with go fmt call