Updated for assignment 1
Updated for assignment 1