fixed for only 1 or 2 arguments
Merge pull request #7 from DustinPham/master

fixed for only 1 or 2 arguments