Removed test code that isn't needed