Updated to use Flat UI Kit