Minor tweaks ;)