Add first part of week4 hw