Got rid of test code