added arguements to the code