Clean out test code