added arguments back to main function