just some small tweaks