Refactor win test after teacher code which makes alot more sense.