Added button to be more responsive.