Add main func
i
Fix main func