Added button to fix problems