Fixing more issues.

Signed-off-by: Ben Corlett <bencorlett@me.com>
Few final tweaks.

Signed-off-by: Ben Corlett <bencorlett@me.com>