small update on layout