modification to allow arguments