Fixed bug - not storing changes