added bunch more tests, TODO fix n_colors