couple small tweaks :D