Removed the test code

Removed the test code from the end of the file