Add run() function with parameter specified in problem