add weight save and readme