Added some code for classes, but it is not connected to the main code.