added a bit of functionality