fix for me, and command line

fixed a bug in my github repo, and added command line argument to just
use command line