added a bit of stuff