add a bit logic to save load method (not functional yet)