Use argparse in conversuncomp.py
Add option to only use the top n layers.