UI + backend bug fixes