Added return statement in main.