Fixing main function return