Updates UI with flatter layout