Hey. Why don't you argparse for getting arguments for user instead of making them provide raw_input. I think it will be a cleaner way.