Working on save and load ui