updating to take args