Add saving and loading with yaml