Removed unused import statements.
Refactored collection class to be more abstract.