Added assignment three code