Add TODOs + go fmt all