Update to UI, button design changed