Add minor tweaks