added check for colour