Remove `main` function