Added 'import argparse' to 'cheat'