add type annotation to main