add arguement

it's not stupid afte4r all