Fix small bug with button and add reload of page on click