Program now handles args correctly