Slight improvement

Added reset button, cleaned up some code.