Added save importance option to script