Added 'TODO:'

Added 'TODO' - use logger and argparse