Take out test code