modified main function to remove two numbers