Add Flat UI back into project