Small reformating, adding a missing method