Minor changes to run the problem