Update README

Add problem_size parameter to the running instruction in README