UI work for gallery page