HW Assignment 4 - v.2