Added argument support.

Signed-off-by: bcnjr5 <bcnjr5@gmail.com>