simple change in Main function