Added save functionality
Implemented save and buggy load methods