Set code back to working order, time to do test thing