better way of handling the args