The second part of week 1 assignment