Minor updates to management command.