Swtich to argparse
Format