Converted to using argparse, and added coloured output option.