added user argument passing in main function