Removed test code (oops)