More dynamic code. Colors and names could be changed in main.