Added colors for different states