Change restore to be a global function. Also add icon for download