Added color selection based on an pull request of @itiboi