Tweaks made to the code while going throw the presentation.

Signed-off-by: Sean Doughety <seancdougherty@gmail.com>