Removed the test code