finished coding lab4, doesn't work though