First part of week 2 assignment