Added first part of assignment 2 for week 1