[Todo] Give an interface for updating draft