Attempt on saving the states in a different way
new way of loadgin states