Update after new feature, see #52