Added TODO to change argument for run function