adding changes after Lab 5