Update README.md
Add support for overriding main function