Moved to argparse and added legend for colums values