Unfucked the code a little bit
cleaned up output