Updating style as per HW4