got color function working