add my own bit of code