have to correct the last print function