Command line arguments now working