Add color differentiation to Epoch in figures