updated for HW6