Revert "Revert "Update main.c""

This reverts commit 95d9c8bfac0995795f43899b9f4f8ec2c35a86f3.