Switch back to argparse, fix/improve some stuff

Switch to argparse, make path optional, fix counting method