Polish result output
Add save/load feature