Removing some test code