Refactored code slightly.

Signed-off-by: retep998 <retep998@verizon.net>