Correct class name
Same fix for other CLI script
Use if/else block
Use if/else block
Merge pull request #49 from wilsonge/patch-1

Correct class name