Button updates properly