Removed some extra test code