Cleaned up FlatUI setup for easier customization