Minor tweaks

Looks great!