Add small tweaks