Implement restore

Change-Id: If7579f73ef0ff69134cc75389125c0436a42ddcc