Readding testing code