Implemented update on model saving