Working version, need fix on main.go for params