Switched to HashMultimap
Fixed weird indents
Fixed wrong import
Fixed various minor things