Add flag to deal with cmd line input

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