Added restart Icon and win Text. Refractored code