Tweaks + fixes