save/update methods working correctly