edited final argument if statement