Added collections and comparators - assignment 4