Restore original main() function