Update to use icons for demo purposes