Add updated ui