Added choose color functionality