Fix bugs, display result, reset when solved