add more tests, make more colorful