Tweaks and cleanup

Signed-off-by: Matthew Schick <matthew.schick@gmail.com>