Finished assginment 2