Button update

Almost there.