Creating a controller class, and using vectors for easier sorting