Added update feature inside models