-added code for assignment part 1