fixed bug with 1 line argument
fixed bug when no args