Update to use chalk rathern than colors