Forgot to add button-renable code