Update main.go

Removed some TODOs