adding a little more functionality