Currently using attributes so I don't have to change that much logic to make it work. Make a more dedicated flow.