Add reset button and update layout