Removing test code