Added assignment solution to stub