Added in code

Code for assignment.