Add progress function
Remove main function