just some small improvements