Accept input as argument