adding restore fature