Completing assignment code

Completing assignment code