[lw2] add saving. now lab #2 is finished