added code for the first class