made some small tweaks