Add cli functionality