Add gradient support to flags