Got basic test going, but doesn't reset for test 2