made updates to make output pretty

T-Pot you've got my back ^_^
Assignment 2 (Final)