Fix main function in this branch.