added start button and end game screen