Add first part of week4 assignment