Removing test code from file