Alternatively, you can put different icon for each. I can already see some difference with the stage one that is squared instead of rounded.