Now accepts command line arguments

Should work with command line arguments now. Still no fix for test file
2.