added tests, victory screen