Small code modifications.

Signed-off-by: Ryan Jackman <ryan1271@verizon.net>