Adding Collector class to handle models logic