wrote code for assignment
wrote code for assignment