added win screen and funtionality