Made choosing a color more clear
Formatted code, added better error messaging