UI changes + schema