Add code to update vertices