Added assignment 2 code