Fix formatting
Add type hint to speed up go-loop