Removed unused class import