added the classes from the previous projet