updated test for new prompt format
more colours