tighten up user prompt code in update function