Added colour as parameter to visualiser