add some code to make it slightly easier to use