Created a class to handle all data