Reverting back

Reverting back to original code.