- Add the main function
- Correct source file name