Remove testing code