Updated to use comand line argument