remove hard coded parameter
do a loop inside run function