Update management command to consider second param