updated UI to flat