Add main.go
Remove test.py