Added error check for missing arg