Adding reset button, styles, cleaning up code