works in  minutes now added argparse