Lil' tweaks

Small tweaks