Fixed a tiny bug.
Added a command line parameter.