Created a h-file for the classes