Reverting back to original code