now it takes command line argument