Removed the testing code