Added functionality to allow custom colour schemes.