Reverting UI back to original